package elven.codingit.easysearch.service;

import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.joda.time.DateTime;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;

import elven.codingit.easysearch.dao.word.WordMapper;
import elven.codingit.easysearch.model.dto.IndexSearchDto;
import elven.codingit.easysearch.model.dto.IndexerResult;
import elven.codingit.easysearch.model.entity.word.Word;
import elven.codingit.easysearch.util.IndexHelper;
import elven.codingit.easysearch.util.IndexWordHelper;
import elven.codingit.easysearch.util.IndexerSearch;
import elven.codingit.easysearch.util.IndexerWrite;
import elven.codingit.easyutil.common.Utils;
import elven.codingit.easyutil.date.DateUtils;
import elven.codingit.easyutil.dto.ReturnDto;

/**
 * lucene,word工具类
 * @author liuyy
 * @date 2016年8月31日
 */
@Service
public class IndexWordService {
	
	private static final String regexStr = "[^\u4E00-\u9FA5A-Za-z0-9 ]";//效验字符串，保留中文英文数字空白
	private static final Logger log = LoggerFactory.getLogger(IndexWordService.class);
	
	@Autowired
	private WordMapper indexWordDao;
	/**
	 * 统计在分类登记了的单词数量，且该分类必须是启用的。
	 * @author liuyy
	 * @date 2016年9月4日
	 */
	public int countWriteWordSize(){
		return indexWordDao.countWriteWordSize();
	}
	
	/**
	 * 统计要写入到索引的单词总数量，即：在有效分类下去重后的单词总数量
	 * @author liuyy
	 * @date 2016年9月4日
	 */
	public int countWriteIndex(){
		return indexWordDao.countWriteIndex();
	}
	
	/**
	 * 对已经在分类中登记的单词查询，并按分类的顺序进行升序排序 只返回单词的列表，如果start,rows没配置，返回所有
	 * @param start 从几开始
	 * @param rows 获取条数
	 * @throws Exception
	 * @author liuyy
	 * @date 2016年9月4日
	 */
	private List<String> distinctContent(int start, int rows) throws Exception{
		Map<String, Object> params = new HashMap<>();
		params.put("start", start);
		params.put("rows", rows);
		return indexWordDao.distinctContent(params);
	}
	
	/**
	 * 根据单词从数据库中获取出有效的单词，排序好并写入到lucene中
	 * @param start 从第几条获取
	 * @param rows 条数
	 * @throws Exception
	 * @author liuyy
	 * @date 2016年9月4日
	 */
	private int contentWrite(int start, int rows) throws Exception{
		log.warn("查询参数limit {}, {}", start, rows);
		List<String> contents =  distinctContent(start, rows);
		List<Word> words = indexWordDao.sortWordByINContent(contents);
		log.warn("当次批量查询出的单词列表：{}", JSON.toJSONString(contents));
		int wordSize = words.size();
		log.warn("根据单词批量查出的总数 : {}", wordSize);
		List<Map<String, Object>> writes = new ArrayList<>();
		for(String key : contents){//遍历单词列表
			writes.add(IndexWordHelper.getIndexWordByDatas(key, words));
		}
		log.warn("写入到索引的总数量: {}", writes.size());
		IndexerWrite indexerWrite = new IndexerWrite();
		indexerWrite.writeApp(writes);
		return wordSize;
	}
	
	/**
	 * 写入单词索引,优化后的写入
	 * 返回总记录数，开始时间结束时间等
	 * @author liuyy
	 * @date 2016年8月31日
	 */
	public ReturnDto updatePaowordIndex() throws Exception{
		log.warn("删除旧的索引文件：{}",IndexHelper.getInstance().removePaowordDirectory());
		JSONObject cont = new JSONObject();
		DateTime start = DateTime.now();
		cont.put("startDate", DateUtils.getCurrent());
		int wordCount = countWriteWordSize();//单词总数量
		int count = countWriteIndex();//写入到索引文件的总条数
		cont.put("wordCount", wordCount);
		cont.put("count", count);
		
		int writeSize = 0;
		int i = 0;
		for(; i < count; i= i+1000){
			writeSize += contentWrite(i, 1000);
		}
		if(i > count){
			writeSize += contentWrite(i - 1000, 1000);
		}
		log.warn("write size:{}, total size : {}", writeSize, count);
		cont.put("endDate", DateUtils.getDetailCurrent());
		return Utils.returnDto(true).content(cont).build();
	}
	
	/**
	 * 根据关键词去查询
	 * @param keyword 关键词
	 * @param rows 影响行数
	 * @throws Exception
	 * @author liuyy
	 * @date 2016年8月31日
	 */
	public IndexerResult findbyKeyword(String keyword, int rows) throws Exception{
		IndexSearchDto searchDto = new IndexSearchDto();
		searchDto.setPage(1);
		searchDto.setRows(rows <= 0 ? 5 : rows);
		searchDto.setKeyword(URLDecoder.decode(keyword, "utf-8"));
		searchDto.setKeyword(searchDto.getKeyword().replaceAll(regexStr, StringUtils.EMPTY));
		searchDto.setFsDirectory(IndexHelper.getInstance().openPaowordDirectory());
		return findbyKeyword(searchDto);
	}
	
	/**
	 * 搜索
	 * @param condtion
	 * @throws Exception
	 * @author liuyy
	 * @date 2016年9月7日
	 */
	public IndexerResult findbyKeyword(IndexSearchDto condtion) throws Exception{
		IndexerResult indexResult =  null;
		if(condtion.getSearchType() == 1){
			condtion.setPage(1);
			condtion.setRows(1);
			indexResult = IndexerSearch.getInstance().termIdQuery(condtion);
		}else{
			indexResult = IndexerSearch.getInstance().search(condtion);
		}
		List<Map<String, Object>> getClientData = new ArrayList<>();
		List<Map<String, Object>> searchdata = indexResult.getResult();
		log.warn("{}", JSON.toJSONString(searchdata));
		for(int i = 0; i < searchdata.size(); i++){
			getClientData.add(IndexWordHelper.mergeAnalyze(searchdata.get(i)));
		}
		indexResult.setResult(getClientData);
		return indexResult;
	}
	
}
